#Data Formating train_idx, test_idx = train_test_split( np.arange(len(y)), test_size = 0.2, shuffle=True, stratify=y, random_state=123) transform = tr.Compose([tr.ToTensor(), tr.Resize((ImageHeight, ImageWidth))]) dataset = ImageFolder(path_images, transform=transform) train_sampler = SubsetRandomSampler(train_idx) test_sampler = SubsetRandomSampler(test_idx) BatchSize=32 train_loader = DataLoader(dataset, batch_size=BatchSize, sampler=train_sampler) test_loader = DataLoader(dataset, batch_size=BatchSize, sampler=test_sampler) dataloader={'train':train_loader,'val':test_loader} #as we don't have a validation set, we took the test set